In [ ]:
fig, ax = plt.subplots(figsize=(8, 3))
d0 = recipient.assign(
withdrawal_due_to_failure=lambda x: x.withdrawal_due_to_failure.astype(float)
).assign(jitter=lambda x: np.linspace(-0.1, 0.1, num=len(x)))
# d0 = recipient.dropna(subset=['mayo_total_start', 'mayo_total_end'])
# x_shift_range = np.linspace(-0.001, 0.001, num=d0.shape[0])
# d0 = d0.sort_values(['mayo_total_start', 'mayo_total_end'], ascending=False).assign(offset=x_shift_range)
plot_kwsA = dict(alpha=1.0)
for subject_id, d1 in d0.iterrows():
if np.isnan(d1.mayo_total_start):
continue
elif d1.withdrawal_due_to_failure == 1.0:
plot_kwsB = dict(linestyle="--", lw=1.5)
d1["mayo_total_end"] = d1["mayo_total_start"]
elif np.isnan(d1.mayo_total_end):
continue
else:
plot_kwsB = dict(linestyle="-", lw=2)
ax.annotate(
"",
xytext=(d1["mayo_total_start"], 0 + d1["jitter"]),
xy=(d1["mayo_total_end"], 1 + d1["jitter"]),
arrowprops=dict(
arrowstyle="-|>",
color=lib.project_style.DEFAULT_COLOR_PALETTE[d1["arm"]],
**plot_kwsA,
**plot_kwsB,
),
# # marker=lib.project_style.DEFAULT_MARKER_PALETTE[d1['donor_subject_id']],
# # marker='o',
# color=lib.project_style.DEFAULT_COLOR_PALETTE[d1['arm']],
# label='__nolegend__',
# **plot_kwsA,
# **plot_kwsB,
)
for arm in lib.project_style.ARM_ORDER:
ax.plot(
[],
[],
color=lib.project_style.DEFAULT_COLOR_PALETTE[arm],
label=arm,
lw=2,
**plot_kwsA,
)
# for donor_subject_id in ['D0044', 'D0097', 'D0485', 'D0065']:
# ax.plot([], [], linestyle='none', marker=lib.project_style.DEFAULT_MARKER_PALETTE[donor_subject_id], color='grey', label=donor_subject_id)
# ax.legend(bbox_to_anchor=(1, 1))
ax.axvline(
2.5,
lw=1,
linestyle="--",
color="grey",
zorder=0,
)
ax.set_xlabel("Mayo Score")
ax.set_yticks([0, 1])
ax.set_ylim(-0.2, 1.2)
ax.set_xlim(-0.5, 10.5)
ax.set_yticklabels(["D0", "F1"])
ax.invert_xaxis()
ax.invert_yaxis()
ax.set_ylabel("← Time")
ax.text(0.2, 0.0, "← worse", fontsize=8, transform=fig.transFigure)
ax.text(0.77, 0.0, "better →", fontsize=8, transform=fig.transFigure)
plt.text(
-0.05, 1.05, "B", fontsize=12, fontweight="heavy", transform=plt.gca().transAxes
)
fig.savefig("fig/mayo_endpoint_arrows.pdf", bbox_inches="tight")
